from pywintypes import Time  # 可以忽视这个 Time 报错（运行程序还是没问题的）
from win32file import GENERIC_READ, GENERIC_WRITE, OPEN_EXISTING
from win32file import CreateFile, SetFileTime, GetFileTime, CloseHandle
# from PIL import Image, ExifTags
import exifread
import os, time
from datetime import datetime
from tkinter import filedialog

# 方法：修改目标文件的时间为源文件时间
def modify_file_time(input, target):
    input_path = input_folder +'/'+ input
    target_path = input_folder +'/' + target
    # TODO: 获取源文件时间
    ctime = os.path.getctime(input_path) #创建时间
    ctime_string = datetime.fromtimestamp(int(ctime))
    mtime = os.path.getmtime(input_path) #修改时间
    mtime_string = datetime.fromtimestamp(int(mtime))
    atime = os.path.getatime(input_path) #修改时间
    atime_string = datetime.fromtimestamp(int(atime))
    file = open(input_path, 'rb')
    exif_tags = exifread.process_file(file) #Exif信息

    print(f"创建时间：{ctime_string}", f"修改时间：{mtime_string}", f"访问时间：{atime_string}",f"EXIF 信息：", sep="\n")
    for tag, value in exif_tags.items():
        print(f"{tag}: {value}")
    # 开始更改操作
    fh = CreateFile(target_path, GENERIC_READ | GENERIC_WRITE, 0, None, OPEN_EXISTING, 0, 0)
    createTimes, accessTimes, modifyTimes = GetFileTime(fh)

    createTimes = Time(ctime)
    accessTimes = Time(atime)
    modifyTimes = Time(mtime)
    # 取最早时间为准
    minTime = min(ctime, atime, mtime)
    if minTime > 0:
        createTimes = Time(minTime)
        modifyTimes = Time(minTime)
    SetFileTime(fh, createTimes, accessTimes, modifyTimes)
    CloseHandle(fh)
    return

# 先让用户选择进行操作的文件夹，识别格式
input_folder = filedialog.askdirectory(title='请选择进行操作的文件夹')
print(input_folder)
input_fomat = input('请输入识别的文件格式：')
output_format = input('请输入输出的修改格式：')

# 遍历文件夹中的所有对应格式的文件（源，目标）
for root,dirs,files in os.walk(input_folder):
    ### 暂时先实现不搜索子文件夹的功能
    # for dir in dirs:
    #     print(os.path.join(root, dir))
    #     # if os.path.join(dir) in ignore:
    #     #     print('撞到忽略目录了，跳过 >v<')
    #     #     break
    #     # else:
    #     os.chdir(os.path.join(root, dir))
    #     time.sleep(0.05)
    # 遍历文件夹中的文件
    input_file = ''
    target_file = ''
    for file in files:
        # print(file)
        if file.endswith('.' + input_fomat):
            print('input file:', file)
            input_file = os.path.splitext(file)[0]
        if file.endswith('.' + output_format):
            print('target file:', file)
            target_file = os.path.splitext(file)[0]
        if (input_file == target_file):
            print('找到匹配的文件:', target_file)
            # TODO: 调用一个方法，修改目标文件的时间为源文件时间
            modify_file_time(input_file+'.'+input_fomat, target_file+'.'+output_format)
        time.sleep(0.05)



# 测试 / 尝试部分
# print('文件夹：' + input_folder, 'input - *.' +
#       input_fomat, '; output - *.' + output_format)


def modifyFileTime(filePath, createTime, modifyTime, accessTime, offset):
    """
    用来修改任意文件的相关时间属性，时间格式：YYYY-MM-DD HH:MM:SS 例如：2019-02-02 00:01:02
    :param filePath: 文件路径名
    :param createTime: 创建时间
    :param modifyTime: 修改时间
    :param accessTime: 访问时间
    :param offset: 时间偏移的秒数,tuple格式，顺序和参数时间对应
    """
    try:
        format = "%Y-%m-%d %H:%M:%S"  # 时间格式
        cTime_t = timeOffsetAndStruct(createTime, format, offset[0])
        mTime_t = timeOffsetAndStruct(modifyTime, format, offset[1])
        aTime_t = timeOffsetAndStruct(accessTime, format, offset[2])

        fh = CreateFile(filePath, GENERIC_READ | GENERIC_WRITE,
                        0, None, OPEN_EXISTING, 0, 0)
        createTimes, accessTimes, modifyTimes = GetFileTime(fh)

        createTimes = Time(time.mktime(cTime_t))
        accessTimes = Time(time.mktime(aTime_t))
        modifyTimes = Time(time.mktime(mTime_t))
        SetFileTime(fh, createTimes, accessTimes, modifyTimes)
        CloseHandle(fh)
        return 0
    except:
        return 1


def timeOffsetAndStruct(times, format, offset):
    return time.localtime(time.mktime(time.strptime(times, format)) + offset)


# if __name__ == '__main__':
#     # 需要自己配置
#     cTime = "2019-12-13 21:51:02"  # 创建时间
#     mTime = "2019-02-02 00:01:03"  # 修改时间
#     aTime = "2019-02-02 00:01:04"  # 访问时间
#     fName = r"E:\test_pro\fileOperate\test.xlsx"  # 文件路径，文件存在才能成功（可以写绝对路径，也可以写相对路径）

#     offset = (0, 1, 2)  # 偏移的秒数（不知道干啥的）

#     # 调用函数修改文件创建时间，并判断是否修改成功
#     r = modifyFileTime(fName, cTime, mTime, aTime, offset)
#     if r == 0:
#         print('修改完成')
#     elif r == 1:
#         print('修改失败')
